# Copyright (c) 2024 Vladimir Botka <vbotka@gmail.com>
# GNU General Public License v3.0+ (see COPYING or https://www.gnu.org/licenses/gpl-3.0.txt)
# SPDX-License-Identifier: GPL-3.0-or-later

from __future__ import annotations

import pytest
import yaml

from ansible.inventory.data import InventoryData
from ansible.template import Templar
from ansible_collections.community.general.plugins.inventory.iocage import InventoryModule
from ansible_collections.community.internal_test_tools.tests.unit.utils.trust import make_trusted


@pytest.fixture
def inventory():
    inv = InventoryModule()
    inv.inventory = InventoryData()
    inv.templar = Templar(None)
    inv.jails = load_txt_data("tests/unit/plugins/inventory/fixtures/iocage/iocage_jails.txt")
    inv.js_ok = load_yml_data("tests/unit/plugins/inventory/fixtures/iocage/iocage_jails.yml")
    inv.jails_dhcp = load_txt_data("tests/unit/plugins/inventory/fixtures/iocage/iocage_jails_dhcp.txt")
    inv.js_dhcp_ok = load_yml_data("tests/unit/plugins/inventory/fixtures/iocage/iocage_jails_dhcp.yml")
    inv.jails_dhcp_nr = load_txt_data("tests/unit/plugins/inventory/fixtures/iocage/iocage_jails_dhcp_not_running.txt")
    inv.js_dhcp_nr_ok = load_yml_data("tests/unit/plugins/inventory/fixtures/iocage/iocage_jails_dhcp_not_running.yml")
    prpts_101 = load_txt_data("tests/unit/plugins/inventory/fixtures/iocage/iocage_properties_test_101.txt")
    prpts_102 = load_txt_data("tests/unit/plugins/inventory/fixtures/iocage/iocage_properties_test_102.txt")
    prpts_103 = load_txt_data("tests/unit/plugins/inventory/fixtures/iocage/iocage_properties_test_103.txt")
    inv.prpts = {"test_101": prpts_101, "test_102": prpts_102, "test_103": prpts_103}
    inv.ps_ok = load_yml_data("tests/unit/plugins/inventory/fixtures/iocage/iocage_properties.yml")
    inv.ok = load_yml_data("tests/unit/plugins/inventory/fixtures/iocage/iocage_inventory.yml")
    return inv


def load_txt_data(path):
    with open(path, "r") as f:
        s = f.read()
    return s


def load_yml_data(path):
    with open(path, "r") as f:
        d = yaml.safe_load(f)
    return d


def get_option(option):
    groups = {}
    groups["test"] = make_trusted("inventory_hostname.startswith('test')")

    if option == "groups":
        return groups
    elif option == "keyed_groups":
        return []
    elif option == "compose":
        return {}
    elif option == "strict":
        return False
    else:
        return None


def test_verify_file_bad_config(inventory):
    assert inventory.verify_file("foobar.iocage.yml") is False


def test_verify_file(tmp_path, inventory):
    file = tmp_path / "foobar.iocage.yml"
    file.touch()
    assert inventory.verify_file(str(file))


def test_get_jails(inventory):
    # jails
    results = {"_meta": {"hostvars": {}}}
    inventory.get_jails(inventory.jails, results)
    assert results == inventory.js_ok

    # jails_dhcp
    results = {"_meta": {"hostvars": {}}}
    inventory.get_jails(inventory.jails_dhcp, results)
    assert results == inventory.js_dhcp_ok

    # jails_dhcp_not_running
    results = {"_meta": {"hostvars": {}}}
    inventory.get_jails(inventory.jails_dhcp_nr, results)
    assert results == inventory.js_dhcp_nr_ok


def test_get_properties(inventory):
    results = {"_meta": {"hostvars": {}}}
    inventory.get_jails(inventory.jails, results)
    for hostname in results["_meta"]["hostvars"]:
        inventory.get_properties(inventory.prpts[hostname], results, hostname)
    assert results == inventory.ps_ok


def test_populate(inventory, mocker):
    results = {"_meta": {"hostvars": {}}}
    inventory.get_jails(inventory.jails, results)
    for hostname in results["_meta"]["hostvars"]:
        inventory.get_properties(inventory.prpts[hostname], results, hostname)
    inventory.get_option = mocker.MagicMock(side_effect=get_option)
    inventory.populate(results)

    # test
    hosts = ("test_101", "test_102", "test_103")
    vars = (
        "iocage_basejail",
        "iocage_boot",
        "iocage_ip4",
        "iocage_ip6",
        "iocage_properties",
        "iocage_release",
        "iocage_state",
        "iocage_template",
        "iocage_type",
    )

    # test host_vars
    for host in hosts:
        h = inventory.inventory.get_host(host)
        for var in vars:
            assert inventory.ok["all"]["children"]["test"]["hosts"][host][var] == h.get_vars()[var]

    # test groups
    test_101_info = inventory.inventory.get_host("test_101")
    test_102_info = inventory.inventory.get_host("test_102")
    test_103_info = inventory.inventory.get_host("test_103")
    g = inventory.inventory.groups["test"]
    assert g.hosts == [test_101_info, test_102_info, test_103_info]
